import pandas as pd
from pandasai import SmartDataframe
from pandasai.llm.openai import OpenAI

# 读取 CSV 文件
df = pd.read_csv('D:/pythondate/.conda/const_prices.csv')

# 检查数据是否为空
if df.empty:
    print("读取的 CSV 文件为空，请检查文件内容。")
else:
    # 将 Date 列转换为日期时间格式
    df['Date'] = pd.to_datetime(df['Date'])

    # 设置 Date 列为索引，同时保留 Date 作为普通列
    df.set_index('Date', inplace=False)
    # 设置 Date 列为索引，同时保留 Date 作为普通列
    df = df.set_index('Date', inplace=False)

    # 配置 DeepSeek 模型
OpenAI._supported_chat_models.append('deepseek-chat')
llm_openai = OpenAI(
        api_base='https://api.deepseek.com',
        api_token='sk-a78b6f4b5cf840319ec23a0671a050df',
        model='deepseek-chat'
    )

# 创建 SmartDataFrame
smart_df = SmartDataframe(df, config={"llm": llm_openai})

# 编写 smart_df.chat 内容以获取每列数据类型
response = smart_df.chat("基于AAPL股票价格数据，"
                        "请分析中美贸易摩擦事件对其影响："
                        "1. 识别2018年3月和2019年5月的股价数据"
                        "2. 比较事件当月与前一个月的价格均值"
                        "3. 分析事件前后对股价的影响"
                        "请用中文简洁回答，使用基础统计方法即可。")
print(response)
